from config import BASE_DIR, SPLIT_CONFIG_FILE
import yaml

with open(BASE_DIR / SPLIT_CONFIG_FILE) as f:
    params = yaml.load(f.read(), Loader=yaml.FullLoader)

src_file = BASE_DIR / params['data']['src']
tgt_file = BASE_DIR / params['data']['tgt']

src_train_file = BASE_DIR / params['output']['src-train']
src_test_file = BASE_DIR / params['output']['src-test']
src_valid_file = BASE_DIR / params['output']['src-valid']

tgt_train_file = BASE_DIR / params['output']['tgt-train']
tgt_test_file = BASE_DIR / params['output']['tgt-test']
tgt_valid_file = BASE_DIR / params['output']['tgt-valid']

shuffle = params['shuffle']
test_radio = params['test_rate']
valid_num = params['valid_num']
